package com.mai4j.maiadmin.support.security;

import com.google.gson.Gson;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import io.jsonwebtoken.impl.DefaultClaims;
import org.apache.commons.codec.digest.DigestUtils;

import java.util.Date;

/**
 * 微信公众号请关注: 开源螺丝钉
 * <br>
 * 码云请关注: <a href="https://gitee.com/xiyoufang">https://gitee.com/xiyoufang</a>
 * <br>
 * 哔哩哔哩请关注: <a href="https://space.bilibili.com/438927834">https://space.bilibili.com/438927834</a>
 *
 * @author xiyoufang
 */
public class JwtUtils {

    /**
     * 创建Token
     *
     * @param key    加密密钥
     * @param value  内容
     * @param expire 失效时间
     * @return Token
     */
    public static String create(String key, Principal value, Date expire) {
        DefaultClaims claims = new DefaultClaims();
        claims.put("data", value);
        return Jwts.builder().setClaims(claims).setExpiration(expire).signWith(SignatureAlgorithm.HS256, DigestUtils.sha256(key)).compact();
    }

    /**
     * 解析
     *
     * @param key   密钥
     * @param token Token
     * @return 解析
     */
    public static Object parse(String key, String token) {
        return Jwts.parser()
                .setSigningKey(DigestUtils.sha256(key))
                .parseClaimsJws(token).getBody().get("data");
    }

    /**
     * 解析
     *
     * @param key   密钥
     * @param token Token
     * @param clazz 类型
     * @param <T>   T
     * @return T
     */
    public static <T extends Principal> T parse(String key, String token, Class<T> clazz) {
        Gson gson = new Gson();
        return gson.fromJson(gson.toJson(Jwts.parser()
                .setSigningKey(DigestUtils.sha256(key))
                .parseClaimsJws(token).getBody().get("data")), clazz);
    }

    /**
     * 校验Token是否有效
     * （包括签名和时效）
     *
     * @param key   密钥
     * @param token Token
     * @return boolean
     */
    public static boolean verify(String key, String token) {
        try {
            parse(key, token);
        } catch (Exception e) {
            return false;
        }
        return true;
    }
}
